package org.bigdata.framework.core.interceptor;

import java.util.List;

import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.bigdata.framework.core.utils.MD5;
import org.bigdata.framework.core.model.User;
import org.bigdata.framework.core.model.UserRole;
import org.bigdata.framework.core.protocol.IPublicService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

/**
 * 拦截器
 * 
 * @author vimes 2016-8-4 19:10:04
 * 
 */
public class SpringMVCInterceptor implements HandlerInterceptor {

	@Autowired
	private IPublicService publicService;

	/**
	 * preHandle方法是进行处理器拦截用的，顾名思义，该方法将在Controller处理之前进行调用，
	 * SpringMVC中的Interceptor拦截器是链式的，可以同时存在
	 * 多个Interceptor，然后SpringMVC会根据声明的前后顺序一个接一个的执行
	 * ，而且所有的Interceptor中的preHandle方法都会在
	 * Controller方法调用之前调用。SpringMVC的这种Interceptor链式结构也是可以进行中断的
	 * ，这种中断方式是令preHandle的返 回值为false，当preHandle的返回值为false的时候整个请求就结束了。
	 */
	@Override
	public boolean preHandle(HttpServletRequest request,
			HttpServletResponse response, Object handler) throws Exception {
		// TODO Auto-generated method stub
		String url = request.getRequestURI();
		if(url.contains("/app-api/") || url.contains("swagger-ui.html")){
			return true;
		}
		if(url.contains("/user-add.html") || url.contains("UserPanDuan.do")
				|| url.contains("userSave.do") || url.contains("ueditorController.do") 
				|| url.contains("/utf8-jsp/")){
			return true;
		}
		// System.out.println(url);
		String usernameCookie = null;
		String passwordCookie = null;
		Cookie[] cookies = request.getCookies();
		if (cookies != null) {
			for (Cookie cookie : cookies) {
				if ("userName".equals(cookie.getName())) {
					usernameCookie = cookie.getValue(); // 得到cookie的用户名
				}
				if ("userPass".equals(cookie.getName())) {
					passwordCookie = cookie.getValue(); // 得到cookie的密码
				}
			}
		}
		if (usernameCookie != null && passwordCookie != null) { // 如果存在
			String password = MD5.convert(passwordCookie);
			List<User> list = publicService.getUser();
			for (int i = 0; i < list.size(); i++) {
				User u = list.get(i);
				if (u.getUsername().equals(usernameCookie)
						&& u.getPassword().equals(password)) {
					request.getSession().setAttribute("username",
							u.getNickName());
					request.getSession()
							.setAttribute("userkk", u.getNickName());
					request.getSession().setAttribute("userid", u.getId());
					request.getSession().setAttribute("kaptchaTime", 0);
					
					List<UserRole> userRole = publicService.getUserRole();
					for (UserRole userRole2 : userRole) {
						if(userRole2.getUserId() == u.getId()){
							if(userRole2.getRoleId()==1){
								request.getSession().setAttribute("userType","超级管理员");
							}else if(userRole2.getRoleId()==2){
								request.getSession().setAttribute("userType","管理员");
							}else if(userRole2.getRoleId()==3){
								request.getSession().setAttribute("userType","教师");
							}
						}
					}
					break;
				}
			}
		}
		if (url.endsWith(".do")) {

			String menu_url = url.split("/")[url.split("/").length - 1];
			// System.out.println(menu_url);
			if (menu_url.equals("userlogin.do")) {

			} else {
				if (request.getSession().getAttribute("username") == null
						|| request.getSession().getAttribute("userid") == null) {
					response.sendRedirect("index.html");
				}
			}
		}
		if (url.endsWith(".html")) {
			String menu_url = url.split("/")[url.split("/").length - 1];
			// System.out.println("menu_url=" + menu_url);
			if (!"login.html".equals(menu_url) && !"login2.html".equals(menu_url) && !"index.html".equals(menu_url)) {
				if (request.getSession().getAttribute("username") == null
						|| request.getSession().getAttribute("userid") == null) {
					response.sendRedirect("index.html");
				}
			}
			
			String[] menu_urlAttay = menu_url.split("-");
			if (menu_urlAttay.length > 2) {
				request.getSession().setAttribute("menu_first",
						menu_urlAttay[0]);
				request.getSession().setAttribute("menu_second", menu_url);
			} else {
				request.getSession().setAttribute("menu_first", menu_url);
			}

		}
		return true;
	}

	/**
	 * 这个方法只会在当前这个Interceptor的preHandle方法返回值为true的时候才会执行。postHandle是进行处理器拦截用的，
	 * 它的执行时间是在处理器进行处理之
	 * 后，也就是在Controller的方法调用之后执行，但是它会在DispatcherServlet进行视图的渲染之前执行
	 * ，也就是说在这个方法中你可以对ModelAndView进行操
	 * 作。这个方法的链式结构跟正常访问的方向是相反的，也就是说先声明的Interceptor拦截器该方法反而会后调用
	 * ，这跟Struts2里面的拦截器的执行过程有点像，
	 * 只是Struts2里面的intercept方法中要手动的调用ActionInvocation的invoke方法
	 * ，Struts2中调用ActionInvocation的invoke方法就是调用下一个Interceptor
	 * 或者是调用action，然后要在Interceptor之前调用的内容都写在调用invoke之前
	 * ，要在Interceptor之后调用的内容都写在调用invoke方法之后。
	 */
	@Override
	public void postHandle(HttpServletRequest request,
			HttpServletResponse response, Object handler,
			ModelAndView modelAndView) throws Exception {
		// TODO Auto-generated method stub

	}

	/**
	 * 该方法也是需要当前对应的Interceptor的preHandle方法的返回值为true时才会执行。该方法将在整个请求完成之后，
	 * 也就是DispatcherServlet渲染了视图执行，
	 * 这个方法的主要作用是用于清理资源的，当然这个方法也只能在当前这个Interceptor的preHandle方法的返回值为true时才会执行。
	 */
	@Override
	public void afterCompletion(HttpServletRequest request,
			HttpServletResponse response, Object handler, Exception ex)
			throws Exception {
		// TODO Auto-generated method stub

	}

}
